absl-py>=0.7.0
jax>=0.2.0
jaxlib>=0.1.69
scikit-learn>=0.23.2